package com.wwz.datasource.config;


public class DynamicDataSourceContextHolder {

    /**
     * 动态数据源名称上下文
     */
    public static final ThreadLocal<String> threadLocal = new ThreadLocal<>();

    /**
     * 设置数据源
     *
     * @param key
     */
    public static void setContextKey(String key) {
        threadLocal.set(key);
    }

    /**
     * 获取数据源
     *
     * @return
     */
    public static String getContextKey() {
        return threadLocal.get();
    }

    /**
     * 删除当前数据源名称
     */
    public static void removeContextKey() {
        threadLocal.remove();
    }

}